package com.yanliang.client;

import com.yanliang.model.RpcRequest;
import com.yanliang.serializer.JSONSerializer;
import com.yanliang.serializer.RpcEncoder;
import io.netty.bootstrap.Bootstrap;
import io.netty.channel.ChannelInitializer;
import io.netty.channel.ChannelOption;
import io.netty.channel.ChannelPipeline;
import io.netty.channel.EventLoopGroup;
import io.netty.channel.nio.NioEventLoopGroup;
import io.netty.channel.socket.SocketChannel;
import io.netty.channel.socket.nio.NioSocketChannel;
import io.netty.handler.codec.string.StringDecoder;

import java.lang.reflect.InvocationHandler;
import java.lang.reflect.Method;
import java.lang.reflect.Proxy;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;

/**
 * @author yanliang
 * @date 5/23/20202:37 PM
 */
public class RpcConsumer {

	//创建线程池对象
	private static ExecutorService executor = Executors.newFixedThreadPool(Runtime.getRuntime().availableProcessors());

	private static UserClientHandler userClientHandler;

	//1.创建一个代理对象
	public Object createProxy(final Class<?> serviceClass,final String className){
		//借助JDK动态代理生成代理对象
		return  Proxy.newProxyInstance(Thread.currentThread().getContextClassLoader(), new Class<?>[]{serviceClass}, new InvocationHandler() {
			public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
				// 调用初始化netty客户端的方法
				if(userClientHandler == null){
					initClient();
				}

//				// 设置参数
//				userClientHandler.setPara(providerName+args[0]);

				// 设置RpcRequest参数
				RpcRequest rpcRequest = RpcRequest.builder()
						.className(className)
						.methodName(method.getName())
						.parameters(args)
						.parameterTypes(method.getParameterTypes())
						.build();
				userClientHandler.setRpcPara(rpcRequest);

				// 去服务端请求数据
				return executor.submit(userClientHandler).get();
			}
		});
	}


	// 2. 初始化netty客户端
	public static void initClient() throws InterruptedException {
		userClientHandler = new UserClientHandler();
		EventLoopGroup group = new NioEventLoopGroup();
		Bootstrap bootstrap = new Bootstrap();
		bootstrap.group(group)
				.channel(NioSocketChannel.class)
				.option(ChannelOption.TCP_NODELAY, true)
				.handler(new ChannelInitializer<SocketChannel>() {
					protected void initChannel(SocketChannel ch) throws Exception {
						ChannelPipeline pipeline = ch.pipeline();
						pipeline.addLast(new StringDecoder());
						pipeline.addLast(new RpcEncoder(RpcRequest.class, new JSONSerializer()));
						pipeline.addLast(userClientHandler);
					}
				});
		bootstrap.connect("127.0.0.1", 8990).sync();

	}
}
